project(1.3-cnn-and-dynamic-shape)

set(BASE_PATH /media/chyang/data/third_party/linux)
set(CUDA_HOME ${BASE_PATH}/cuda/cuda_12.3)
set(TRT_HOME ${BASE_PATH}/tensorrt/TensorRT-8.6.1.6/)

# Find CUDA.
include_directories(${CUDA_HOME}/include)
link_directories(${CUDA_HOME}/lib)

# Find TensorRT
# include_directories(${TRT_HOME}/include/)
link_directories(${TRT_HOME}/lib)

include_directories(${CMAKE_CURRENT_SOURCE_DIR}/src)

cuda_add_library(myplugins SHARED ${CMAKE_CURRENT_SOURCE_DIR}/src/myselu-kernel.cu)

add_executable(guide src/main.cpp)
target_link_libraries(guide PUBLIC cuda cudart nvinfer nvinfer_plugin nvonnxparser myplugins)